universal-mcp-agents 0.1.23rc10__py3-none-any.whl → 0.1.23rc11__py3-none-any.whl

universal_mcp/agents/base.py

@@ -67,6 +67,8 @@ class BaseAgent:
             ):
                 if event == "messages" and isinstance(meta, (tuple, list)) and len(meta) == 2:  # noqa: PLR2004
                     payload, meta_dict = meta
+                    if meta_dict.get("tags") and "quiet" in meta_dict.get("tags"):
+                        continue
                     is_agent_builder = (
                         isinstance(meta_dict, dict) and meta_dict.get("langgraph_node") == "agent_builder"
                     )

universal_mcp/agents/cli.py

@@ -18,14 +18,14 @@ app = Typer()
     mcp client run --config client_config.json
     """,
 )
-def run(name: str = "react"):
+def run(name: str = "codeact-repl"):
     """Run the agent CLI"""
 
     setup_logger(log_file=None, level="ERROR")
     client = AgentrClient()
     params = {
         "instructions": "You are a helpful assistant",
-        "model": "azure/gpt-4.1",
+        "model": "anthropic:claude-4-sonnet-20250514",
         "registry": AgentrRegistry(client=client),
         "memory": MemorySaver(),
     }

universal_mcp/agents/codeact0/agent.py

@@ -27,7 +27,7 @@ from universal_mcp.agents.codeact0.tools import (
     create_meta_tools,
     enter_agent_builder_mode,
 )
-from universal_mcp.agents.codeact0.utils import build_anthropic_cache_message, get_connected_apps_string
+from universal_mcp.agents.codeact0.utils import build_anthropic_cache_message, get_connected_apps_string, strip_thinking
 from universal_mcp.agents.llm import load_chat_model
 from universal_mcp.agents.utils import convert_tool_ids_to_dict, filter_retry_on, get_message_text
 
@@ -52,7 +52,7 @@ class CodeActPlaybookAgent(BaseAgent):
             **kwargs,
         )
         self.model_instance = load_chat_model(model)
-        self.agent_builder_model_instance = load_chat_model("anthropic:claude-sonnet-4-5-20250929")
+        self.agent_builder_model_instance = load_chat_model("anthropic:claude-sonnet-4-5-20250929", thinking = False)
         self.registry = registry
         self.agent_builder_registry = agent_builder_registry
         self.agent = agent_builder_registry.get_agent() if agent_builder_registry else None
@@ -217,7 +217,7 @@ class CodeActPlaybookAgent(BaseAgent):
                 plan_id = str(uuid.uuid4())
                 writer({"type": "custom", id: plan_id, "name": "planning", "data": {"update": bool(self.agent)}})
                 planning_instructions = self.instructions + AGENT_BUILDER_PLANNING_PROMPT + self.preloaded_defs
-                messages = [{"role": "system", "content": planning_instructions}] + state["messages"]
+                messages = [{"role": "system", "content": planning_instructions}] + strip_thinking(state["messages"])
 
                 model_with_structured_output = self.agent_builder_model_instance.with_structured_output(
                     AgentBuilderPlan

universal_mcp/agents/codeact0/prompts.py

@@ -6,13 +6,13 @@ uneditable_prompt = """
 You are **Ruzo**, an AI Assistant created by AgentR — a creative, straight-forward, and direct principal software engineer with access to tools.
 
 Your job is to answer the user's question or perform the task they ask for.
-- Answer simple questions (which do not require you to write any code or access any external resources) directly. Note that any operation that involves using ONLY print functions should be answered directly in the chat. NEVER write a string yourself and print it.
+- Answer simple questions (which do not require you to write any code or access any external resources) directly. Note that any operation that involves using ONLY print functions should be answered directly in the chat. NEVER write a string or sequences of strings yourself and print it.
 - For task requiring operations or access to external resources, you should achieve the task by executing Python code snippets.
 - You have access to `execute_ipython_cell` tool that allows you to execute Python code in an IPython notebook cell.
+- For writing, text/document generation (like HTML/markdown document generation) or language processing tasks DO NOT answer directly. Instead you MUST use `execute_ipython_cell` tool with the llm functions provided to you for tasks like summarizing, text generation, classification, data extraction from text or unstructured data, etc. Avoid hardcoded approaches to classification, data extraction, or creative writing.
 - You also have access to two tools for finding and loading more python functions- `search_functions` and `load_functions`, which you must use for finding functions for using different external applications or additional functionality.
     - Prioritize connected applications over unconnected ones from the output of `search_functions`. However, if the user specifically asks for an application, you MUST use that irrespective of connection status.
     - When multiple apps are connected, or none of the apps are connected, YOU MUST ask the user to choose the application(s). The search results will inform you when such a case occurs, and you must stop and ask the user if multiple apps are relevant.
-- In writing or natural language processing tasks DO NOT answer directly. Instead use `execute_ipython_cell` tool with the AI functions provided to you for tasks like summarizing, text generation, classification, data extraction from text or unstructured data, etc. Avoid hardcoded approaches to classification, data extraction, or creative writing.
 - The code you write will be executed in a sandbox environment, and you can use the output of previous executions in your code. variables, functions, imports are retained.
 - Read and understand the output of the previous code snippet and use it to answer the user's request. Note that the code output is NOT visible to the user, so after the task is complete, you have to give the output to the user in a markdown format. Similarly, you should only use print/smart_print for your own analysis, the user does not get the output.
 - If needed, feel free to ask for more information from the user (without using the `execute_ipython_cell` tool) to clarify the task.
@@ -21,14 +21,14 @@ Your job is to answer the user's question or perform the task they ask for.
 - The code you write will be executed in a sandbox environment, and you can use the output of previous executions in your code. Variables, functions, imports are retained.
 - Read and understand the output of the previous code snippet and use it to answer the user's request. Note that the code output is NOT visible to the user, so after the task is complete, you have to give the output to the user in a markdown format. Similarly, you should only use print/smart_print for your own analysis, the user does not get the output.
 - If needed, feel free to ask for more information from the user (without using the `execute_ipython_cell` tool) to clarify the task.
-- Always describe in 2-3 lines about the current progress. In each step, mention what has been achieved and what you are planning to do next.
-- DO NOT use the code execution to communicate with the user. The user is not able to see the output of the code cells.
+- DO NOT use the code execution to communicate/show anything to the user. The user is not able to see the output of the code cells, it is only for your processing and actions.
+- Whenever you need to generate a large body of text, such as a document, an HTML file, or a report, use llm functions and save the output file using save functions. Do not generate text yourself and do not print the entire text in order to save your memory.
 
 **Coding Best Practices:**
-- Variables defined at the top level of previous code snippets can be referenced in your code.
+- Structure your code into multiple small, well-defined functions within a single execution snippet. This ensures modularity and makes it easier to debug or update specific logic without rewriting or re-executing large portions of code. You can only rewrite the function/portion that you need to edit since the others are retained in context.
+- Variables, functions, classes defined at the top level of previous code snippets can be referenced in your code.
 - External functions which return a dict or list[dict] are ambiguous. Therefore, you MUST explore the structure of the returned data using `smart_print()` statements before using it, printing keys and values. `smart_print` truncates long strings from data, preventing huge output logs.
 - When an operation involves running a fixed set of steps on a list of items, run one run correctly and then use a for loop to run the steps on each item in the list.
-- In a single code snippet, try to achieve as much as possible.
 - You can only import libraries that come pre-installed with Python. However, do consider searching for external functions first, using the search and load tools to access them in the code.
 - For displaying final results to the user, you must present your output in markdown format, including image links, so that they are rendered and displayed to the user. The code output is NOT visible to the user.
 - Call all functions using keyword arguments only, never positional arguments.
@@ -39,7 +39,7 @@ Your job is to answer the user's question or perform the task they ask for.
     - Anything that's just formatted print statements
 
 **Final Output Requirements:**
-- Once you have all the information about the task, return the text directly to user in markdown format. Do NOT call `execute_ipython_cell` again just for summarization.
+- Once you have all the information about the task, return the text directly to user in markdown format. Do NOT call `execute_ipython_cell` or any LLM tools again just for summarization. Do NOT use llm__generate_text for this purpose.
 - Always respond in github flavoured markdown format.
 - For charts and diagrams, use mermaid chart in markdown directly.
 - Your final response should contain the complete answer to the user's request in a clear, well-formatted manner that directly addresses what they asked for.
@@ -209,9 +209,7 @@ def create_default_prompt(
         system_prompt = uneditable_prompt.strip()
         if apps_string:
             system_prompt += f"\n\n**Connected external applications (These apps have been logged into by the user):**\n{apps_string}\n\n Use `search_functions` to search for functions you can perform using the above. You can also discover more applications using the `search_functions` tool to find additional tools and integrations, if required. However, you MUST not assume the application when multiple apps are connected for a particular usecase.\n"
-        system_prompt += (
-            "\n\nIn addition to the Python Standard Library, you can use the following external functions:\n"
-        )
+        system_prompt += "\n\nIn addition to the Python Standard Library, you can use the following external functions:\n Carefully note which functions are normal and which functions are async. CRITICAL: Use `await` with async functions and async functions ONLY."
     else:
         system_prompt = ""
 

universal_mcp/agents/codeact0/tools.py

@@ -320,7 +320,8 @@ def create_meta_tools(tool_registry: AgentrRegistry) -> dict[str, Any]:
         valid_tools, unconnected_links = await get_valid_tools(tool_ids=tool_ids, registry=tool_registry)
 
         if not valid_tools:
-            return "Error: None of the provided tool IDs could be validated or loaded."
+            response_string = "Error: None of the provided tool IDs could be validated or loaded."
+            return response_string, {}, [], ""
 
         # Step 2: Export the schemas of the valid tools.
         await tool_registry.load_tools(valid_tools)
@@ -502,7 +503,7 @@ async def get_valid_tools(tool_ids: list[str], registry: AgentrRegistry) -> tupl
             start = text.find(":") + 1
             end = text.find(". R", start)
             url = text[start:end].strip()
-            markdown_link = f"[{app}]({url})"
+            markdown_link = f"![Connect to {app.capitalize()}]({url})"
             unconnected_links.append(markdown_link)
         for tool_id, tool_name in tool_entries:
             if tool_name in available:

universal_mcp/agents/codeact0/utils.py

@@ -4,7 +4,7 @@ import re
 from collections.abc import Sequence
 from typing import Any
 
-from langchain_core.messages import BaseMessage
+from langchain_core.messages import AIMessage, BaseMessage
 from universal_mcp.types import ToolConfig
 
 MAX_CHARS = 5000
@@ -29,6 +29,56 @@ def build_anthropic_cache_message(text: str, role: str = "system", ttl: str = "1
         }
     ]
 
+def strip_thinking(messages: list[BaseMessage]):
+    """Remove Anthropic 'thinking' segments from the most recent AIMessage in-place.
+
+    Scans from the end to find the last AIMessage, then removes thinking blocks
+    from its content. Handles both plain-string and block-array content.
+    """
+    if not messages:
+        return messages
+
+    # Find the last AIMessage from the end
+    last_ai_index = None
+    for i in range(len(messages) - 1, -1, -1):
+        if isinstance(messages[i], AIMessage):
+            last_ai_index = i
+            break
+
+    if last_ai_index is None:
+        return messages
+
+    ai_msg = messages[last_ai_index]
+    content = ai_msg.content
+
+    # If it's already plain text, nothing to strip
+    if isinstance(content, str):
+        return messages
+
+    # If Anthropic-style content blocks
+    if isinstance(content, list):
+        filtered_output: list[object] = []
+        removed_any = False
+        for b in content:
+            is_thinking = False
+            if isinstance(b, dict):
+                t = b.get("type")
+                if t == "thinking":
+                    is_thinking = True
+                elif "thinking" in b and isinstance(b["thinking"], str):
+                    is_thinking = True
+
+            if is_thinking:
+                removed_any = True
+                continue
+            filtered_output.append(b)
+
+        if removed_any:
+            ai_msg.content = filtered_output
+            messages[last_ai_index] = ai_msg
+
+    return messages
+
 
 def add_tools(tool_config: ToolConfig, tools_to_add: ToolConfig):
     for app_id, new_tools in tools_to_add.items():

universal_mcp/agents/llm.py

@@ -9,22 +9,28 @@ from loguru import logger
 
 @lru_cache(maxsize=8)
 def load_chat_model(
-    fully_specified_name: str, temperature: float = 1.0, tags: list[str] | None = None, thinking: bool = True
+    fully_specified_name: str, temperature: float = 1.0, tags: tuple[str, ...] | None = None, thinking: bool = True, disable_streaming: bool = False
 ) -> BaseChatModel:
     """Load a chat model from a fully specified name.
     Args:
         fully_specified_name (str): String in the format 'provider/model'.
     """
     fully_specified_name = fully_specified_name.replace("/", ":")
+    if tags:
+        if isinstance(tags, str):
+            tags = [tags]
+        else:
+            tags = list[str](tags)
     provider, model = fully_specified_name.split(":", maxsplit=1)
     if provider == "anthropic":
         return ChatAnthropic(
             model=model,
             temperature=temperature,
             thinking={"type": "enabled", "budget_tokens": 2048} if thinking else None,
-            max_tokens=4096,
+            max_tokens=8096,
             tags=tags,
             stream_usage=True,
+            disable_streaming = disable_streaming
         )  # pyright: ignore[reportCallIssue]
     elif provider == "azure":
         return AzureChatOpenAI(
@@ -34,6 +40,7 @@ def load_chat_model(
             temperature=temperature,
             tags=tags,
             stream_usage=True,
+            disable_streaming = disable_streaming
         )
     elif provider == "gemini":
         return ChatGoogleGenerativeAI(model=model, temperature=temperature)

universal_mcp/applications/llm/app.py

@@ -91,7 +91,7 @@ class LlmApp(BaseApplication):
 
         full_prompt = f"{prompt}\n\nContext:\n{context_str}\n\n"
 
-        model = load_chat_model("azure/gpt-5-mini")
+        model = load_chat_model("azure/gpt-5-mini", disable_streaming = True, tags=("quiet",))
         response = model.with_retry(stop_after_attempt=MAX_RETRIES).invoke(full_prompt, stream=False)
         return str(response.content)
 
@@ -158,7 +158,7 @@ class LlmApp(BaseApplication):
             reason: str = Field(..., description="The reasoning behind the classification.")
             top_class: str = Field(..., description="The class with the highest probability.")
 
-        model = load_chat_model("azure/gpt-5-mini", temperature=0)
+        model = load_chat_model("azure/gpt-5-mini", temperature=0, disable_streaming = True, tags=("quiet",))
         agent = create_agent(
             model=model,
             tools=[],
@@ -229,7 +229,7 @@ class LlmApp(BaseApplication):
             "Return ONLY a valid JSON object that conforms to the provided schema, with no extra text."
         )
 
-        model = load_chat_model("azure/gpt-5-mini", temperature=0)
+        model = load_chat_model("azure/gpt-5-mini", temperature=0, disable_streaming = True, tags=("quiet",))
 
         response = (
             model.with_structured_output(schema=output_schema, method="json_mode")
@@ -282,7 +282,7 @@ class LlmApp(BaseApplication):
 
         prompt = f"{task_instructions}\n\nContext:\n{context_str}\n\nReturn ONLY a valid JSON object, no extra text."
 
-        model = load_chat_model("azure/gpt-5-mini", temperature=0)
+        model = load_chat_model("azure/gpt-5-mini", temperature=0, disable_streaming = True, tags=("quiet",))
 
         agent = create_agent(
             model=model,

{universal_mcp_agents-0.1.23rc10.dist-info → universal_mcp_agents-0.1.23rc11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp-agents
-Version: 0.1.23rc10
+Version: 0.1.23rc11
 Summary: Add your description here
 Project-URL: Homepage, https://github.com/universal-mcp/applications
 Project-URL: Repository, https://github.com/universal-mcp/applications
@@ -12,8 +12,8 @@ Requires-Dist: langchain-anthropic>=0.3.19
 Requires-Dist: langchain-google-genai>=2.1.10
 Requires-Dist: langchain-openai>=0.3.32
 Requires-Dist: langgraph>=0.6.6
-Requires-Dist: universal-mcp-applications>=0.1.25
-Requires-Dist: universal-mcp>=0.1.24rc27
+Requires-Dist: universal-mcp-applications>=0.1.30
+Requires-Dist: universal-mcp>=0.1.24rc29
 Provides-Extra: dev
 Requires-Dist: pre-commit; extra == 'dev'
 Requires-Dist: ruff; extra == 'dev'

{universal_mcp_agents-0.1.23rc10.dist-info → universal_mcp_agents-0.1.23rc11.dist-info}/RECORD

@@ -1,8 +1,8 @@
 universal_mcp/agents/__init__.py,sha256=Ythw8tyq7p-w1SPnuO2JtS4TvYEP75PkQpdyvZv-ww4,914
-universal_mcp/agents/base.py,sha256=IyU1HUmB8rjHuCxv-c29RV-dWXfdiQiPq5rGkcCiSbU,7833
-universal_mcp/agents/cli.py,sha256=9CG7majpWUz7C6t0d8xr-Sg2ZPKBuQdykTbYS6KIZ3A,922
+universal_mcp/agents/base.py,sha256=HkTQzh8EudkG4_5teiD0aDidWpfp2obOz0XR9XsZeUI,7949
+universal_mcp/agents/cli.py,sha256=d3O5BxkT4axHcKnL7gM1SvneWoZMh1QQoElk03KeuU4,950
 universal_mcp/agents/hil.py,sha256=_5PCK6q0goGm8qylJq44aSp2MadP-yCPvhOJYKqWLMo,3808
-universal_mcp/agents/llm.py,sha256=S6dI3xaeVS8rKa2ttXToOYf_mI-6rm0E9XwE5nm3uko,1782
+universal_mcp/agents/llm.py,sha256=AOijT3hCt3VId97FnvARsPG9lx0sLmn-yiUo8Qx90lQ,2043
 universal_mcp/agents/react.py,sha256=ocYm94HOiJVI2zwTjO1K2PNfVY7EILLJ6cd__jnGHPs,3327
 universal_mcp/agents/sandbox.py,sha256=YxTGp_zsajuN7FUn0Q4PFjuXczgLht7oKql_gyb2Gf4,5112
 universal_mcp/agents/simple.py,sha256=NSATg5TWzsRNS7V3LFiDG28WSOCIwCdcC1g7NRwg2nM,2095
@@ -22,23 +22,23 @@ universal_mcp/agents/builder/prompts.py,sha256=8Xs6uzTUHguDRngVMLak3lkXFkk2VV_uQ
 universal_mcp/agents/builder/state.py,sha256=7DeWllxfN-yD6cd9wJ3KIgjO8TctkJvVjAbZT8W_zqk,922
 universal_mcp/agents/codeact0/__init__.py,sha256=8-fvUo1Sm6dURGI-lW-X3Kd78LqySYbb5NMkNJ4NDwg,76
 universal_mcp/agents/codeact0/__main__.py,sha256=YyIoecUcKVUhTcCACzLlSmYrayMDsdwzDEqaV4VV4CE,766
-universal_mcp/agents/codeact0/agent.py,sha256=CRgbKCBHSbMOpiNp_Sgdb-Wml7o9Uy72aA9_DaPNiJA,23449
+universal_mcp/agents/codeact0/agent.py,sha256=Hwp7awECd6DPYL39LzGFrsqTkV0uCXWuVEJ_dUHxB-0,23499
 universal_mcp/agents/codeact0/config.py,sha256=H-1woj_nhSDwf15F63WYn723y4qlRefXzGxuH81uYF0,2215
 universal_mcp/agents/codeact0/langgraph_agent.py,sha256=8nz2wq-LexImx-l1y9_f81fK72IQetnCeljwgnduNGY,420
 universal_mcp/agents/codeact0/llm_tool.py,sha256=-pAz04OrbZ_dJ2ueysT1qZd02DrbLY4EbU0tiuF_UNU,798
-universal_mcp/agents/codeact0/prompts.py,sha256=RiC_43GSeE4LDoiFhmJIOsKkoijOK9_7skwAH6ZqSWk,15501
+universal_mcp/agents/codeact0/prompts.py,sha256=02VAmWca0rirdE1d-gBZgKIw4Jg20eciJ9jppyNbFzI,16250
 universal_mcp/agents/codeact0/sandbox.py,sha256=Zcr7fvYtcGbwNWd7RPV7-Btl2HtycPIPofEGVmzxSmE,4696
 universal_mcp/agents/codeact0/state.py,sha256=cf-94hfVub-HSQJk6b7_SzqBS-oxMABjFa8jqyjdDK0,1925
-universal_mcp/agents/codeact0/tools.py,sha256=oaGBzto6yaysPPEwV0bpAHH8QASjEaTIey_zJHxmNyY,23182
-universal_mcp/agents/codeact0/utils.py,sha256=Gvft0W0Sg1qlFWm8ciX14yssCa8y3x037lql92yGsBQ,18164
+universal_mcp/agents/codeact0/tools.py,sha256=Kr8FZ8DyrUhxekbW52bhY-y2EzCj1hzsvWOeyKfB_5Q,23265
+universal_mcp/agents/codeact0/utils.py,sha256=eOdqemLx6RqI5g--vtM9WP2LHrYXro2bxScNAVUU0Do,19683
 universal_mcp/agents/shared/__main__.py,sha256=XxH5qGDpgFWfq7fwQfgKULXGiUgeTp_YKfcxftuVZq8,1452
 universal_mcp/agents/shared/prompts.py,sha256=yjP3zbbuKi87qCj21qwTTicz8TqtkKgnyGSeEjMu3ho,3761
 universal_mcp/agents/shared/tool_node.py,sha256=DC9F-Ri28Pam0u3sXWNODVgmj9PtAEUb5qP1qOoGgfs,9169
 universal_mcp/applications/filesystem/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 universal_mcp/applications/filesystem/app.py,sha256=0TRjjm8YnslVRSmfkXI7qQOAlqWlD1eEn8Jm0xBeigs,5561
 universal_mcp/applications/llm/__init__.py,sha256=_XGRxN3O1--ZS5joAsPf8IlI9Qa6negsJrwJ5VJXno0,46
-universal_mcp/applications/llm/app.py,sha256=y84p2gIIsgTG5A1fxtZCPUAr3k4qacg8bp2jntBxy8A,12761
+universal_mcp/applications/llm/app.py,sha256=2BIH6JDqw6LPD9Iuo3_LoXa48813o6G4gmeRBR_v9oc,12933
 universal_mcp/applications/ui/app.py,sha256=c7OkZsO2fRtndgAzAQbKu-1xXRuRp9Kjgml57YD2NR4,9459
-universal_mcp_agents-0.1.23rc10.dist-info/METADATA,sha256=J2tqsjN2LrLGdhyhjLGKwQHrXWenn2D4rH5Mqp20QWc,932
-universal_mcp_agents-0.1.23rc10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp_agents-0.1.23rc10.dist-info/RECORD,,
+universal_mcp_agents-0.1.23rc11.dist-info/METADATA,sha256=-nhuSFRtXvQJz1HSDb-QxTFAMH399Ve8tYVAqIhLM9c,932
+universal_mcp_agents-0.1.23rc11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp_agents-0.1.23rc11.dist-info/RECORD,,